#!/bin/bash
cuda=2
task=Task607_CZ2
trainer=new_fine_nyunet0102_epoch500
chk=model_best
fold=0
threads_preprocessing=1
num_threads_nifti_save=1

input_folder=/home/yusongli/_dataset/shidaoai/img/_out/nn/DATASET/nnUNet_raw_data_base/nnUNet_raw_data/${task}/val_images
output_folder=/home/yusongli/_dataset/shidaoai/img/_out/nn/DATASET/nnUNet_cropped_data/nnUNet/3d_fullres/${task}/${trainer}__nnUNetPlansv2.1/fold_0/val_preds


CUDA_DEVICE_ORDER=PCI_BUS_ID \
CUDA_VISIBLE_DEVICES=${cuda} \
nnUNet_predict \
--input_folder ${input_folder} \
--output_folder ${output_folder} \
--task_name ${task} \
--trainer_class_name ${trainer} \
--folds ${fold} \
--chk ${chk} \
--save_npz \
--disable_mixed_precision \
--num_threads_preprocessing ${num_threads_preprocessing} \
--num_threads_nifti_save ${num_threads_nifti_save}
